package p100;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * @author pengjz <br>
 * @version 1.0 <br>
 * @description Leetcode_131 <br>
 * @date 2021/3/7 22:36 <br>
 */
public class Leetcode_131 {
    /**
     * dp数组
     */
    boolean[][] dp;
    /**
     * 所有结果
     */
    List<List<String>> ret = new ArrayList<>();
    /**
     * 一条结果
     */
    List<String> ans = new ArrayList<>();
    int n;

    public List<List<String>> partition(String s) {
        n = s.length();
        // 初始化dp数组
        dp = new boolean[n][n];
        for (int i = 0; i < n; i++) {
            Arrays.fill(dp[i], true);
        }
        for (int i = n - 1; i >= 0; --i) {
            for (int j = i + 1; j < n; ++j) {
                dp[i][j] = (s.charAt(i) == s.charAt(j)) && dp[i + 1][j - 1];
            }
        }
        // 根据dp数组递归搜索解
        dfs(s, 0);
        return ret;
    }

    private void dfs(String s, int i) {
        // 遍历到了最后一个字符，说明搜索结束，找到一个解
        if (i == n) {
            ret.add(new ArrayList<>(ans));
        }
        for (int j = i; j < n; j++) {
            // 如果i到j是一个回文字符串
            if (dp[i][j]) {
                // 吧这个字符串添加到解答中
                ans.add(s.substring(i, j + 1));
                // 递归搜索剩下的字符串
                dfs(s, j + 1);
                // 上层递归中已经添加过，本轮需要移除
                ans.remove(ans.size() - 1);
            }
        }
    }

    public static void main(String[] args) {
        Leetcode_131 l = new Leetcode_131();
        System.out.println(l.partition("aab"));
    }
}
